package com.crawler.gmail.crawler;

import java.io.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;
import java.util.zip.*;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author lizengfa
 */
public final class FileUtil {

    private final static Logger LOGGER = LoggerFactory.getLogger(FileUtil.class);

    /**
     * 创建临时文件，默认会在jvm退出后删除所有创建的临时文件，
     * 但最好创建使用后手动删除，以免创建临时文件过多耗光操作系统
     * 文件句柄
     *
     * @param bytes
     * @param suffix
     * @return
     */
    public static File createTempleFile(byte[] bytes, String suffix) {
        File file = null;
        FileOutputStream fos = null;
        try {
            if (!StringUtils.isBlank(suffix)) {
                file = File.createTempFile(UUID.randomUUID().toString(), "." + suffix);
            } else {
                file = File.createTempFile("temp_", UUID.randomUUID().toString());
            }
            file.deleteOnExit();
            fos = new FileOutputStream(file);
            fos.write(bytes);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                fos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return file;
    }

    public static void deleteFile(File file) {
        if (!file.exists()) {
            return;
        }
        if (file.isFile()) {
            file.delete();
        } else {
            File[] files = file.listFiles();
            if (files == null) {
                return;
            }
            for (File f : files) {
                deleteFile(f);
            }
//			File[] files2 = file.listFiles();
//			if (files2.length == 0) {
//				file.delete();
//			}
            file.delete();
        }
    }


    /**
     * 初始化压缩包信息并开始进行压缩
     *
     * @param inputFile  需要压缩的文件或文件夹
     * @param outputFile 压缩后的文件
     */
    public static void zip(File inputFile, File outputFile) {
        ZipOutputStream zos = null;
        try {
            zos = new ZipOutputStream(new FileOutputStream(outputFile));
            // 设置压缩包注释
            zos.setComment("From Log");
            zipFile(zos, inputFile, null);
        } catch (IOException e) {
            LOGGER.info("压缩文件错误{}", e.getMessage(), e);
        } finally {
            if (zos != null) {
                try {
                    zos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 如果是单个文件，那么就直接进行压缩。如果是文件夹，那么递归压缩所有文件夹里的文件
     *
     * @param zos       压缩输出流
     * @param inputFile 需要压缩的文件
     * @param path      需要压缩的文件在压缩包里的路径
     */
    public static void zipFile(ZipOutputStream zos, File inputFile, String path) {
        if (inputFile.isDirectory()) {
            // 记录压缩包中文件的全路径
            String p = null;
            File[] fileList = inputFile.listFiles();
            for (int i = 0; i < fileList.length; i++) {
                File file = fileList[i];
                // 如果路径为空，说明是根目录
                if (path == null || path.isEmpty()) {
                    p = file.getName();
                } else {
                    p = path + File.separator + file.getName();
                }
                if ("lock".equals(file.getName())) {
                    continue;
                }
                // 如果是目录递归调用，直到遇到文件为止
                zipFile(zos, file, p);
            }
        } else {
            zipSingleFile(zos, inputFile, path);
        }
    }

    /**
     * 压缩单个文件到指定压缩流里
     *
     * @param zos       压缩输出流
     * @param inputFile 需要压缩的文件
     * @param path      需要压缩的文件在压缩包里的路径
     * @throws FileNotFoundException
     */
    public static void zipSingleFile(ZipOutputStream zos, File inputFile, String path) {
        try {
            InputStream in = new FileInputStream(inputFile);
            zos.putNextEntry(new ZipEntry(path));
            write(in, zos);
        } catch (IOException e) {
            LOGGER.info("压缩写入文件错误{}", e.getMessage(), e);
        }
    }

    /**
     * 从输入流写入到输出流的方便方法 【注意】这个函数只会关闭输入流，且读写完成后会调用输出流的flush()函数，但不会关闭输出流！
     *
     * @param input
     * @param output
     */
    private static void write(InputStream input, OutputStream output) {
        int len = -1;
        byte[] buff = new byte[1024];
        try {
            while ((len = input.read(buff)) != -1) {
                output.write(buff, 0, len);
            }
            output.flush();
        } catch (IOException e) {
            LOGGER.info("写入文件错误{}", e.getMessage(), e);
        } finally {
            try {
                input.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }


    /**
     * 解压压缩包到指定目录
     *
     * @param inputFile
     * @param outputFile
     */
    public static void unZip(File inputFile, File outputFile) {
        if (!outputFile.exists()) {
            outputFile.mkdirs();
        }

        ZipFile zipFile = null;
        ZipInputStream zipInput = null;
        ZipEntry entry = null;
        OutputStream output = null;
        InputStream input = null;
        File file = null;
        try {
            zipFile = new ZipFile(inputFile);
            zipInput = new ZipInputStream(new FileInputStream(inputFile));
            String path = outputFile.getAbsolutePath() + File.separator;
            while ((entry = zipInput.getNextEntry()) != null) {
                // 从压缩文件里获取指定已压缩文件的输入流
                input = zipFile.getInputStream(entry);

                // 拼装压缩后真实文件路径
                String fileName = path + entry.getName();

                // 创建文件缺失的目录（不然会报异常：找不到指定文件）
                file = new File(fileName.substring(0, fileName.lastIndexOf(File.separator)));
                file.mkdirs();

                // 创建文件输出流
                output = new FileOutputStream(fileName);

                // 写出解压后文件数据
                write(input, output);
                output.close();
            }
        } catch (ZipException e) {
            LOGGER.info("文件解压错误{}", e.getMessage(), e);
        } catch (IOException e) {
            LOGGER.info("文件解压错误{}", e.getMessage(), e);
        } finally {
            try {
                if (output != null) {
                    output.close();
                }

                if (zipInput != null) {
                    zipInput.close();
                }
            } catch (IOException e) {
                LOGGER.info("文件解压错误{}", e.getMessage(), e);
            }
        }
    }

    public final static List<String> cookieFileList = Arrays.asList(
            "SiteSecurityServiceState.txt",
            "cert8.db",
            "cookies.sqlite",
            "cookies.sqlite-shm",
            "cookies.sqlite-wal",
            "key3.db",
            "places.sqlite",
            "places.sqlite-shm",
            "places.sqlite-wal",
            "secmod.db",
            "storage",
            "webappsstore.sqlite",
            "webappsstore.sqlite-wal",
            "webappsstore.sqlite-shm"
    );

    public static void deleteZhilianCookieFile(File file) {
        if (file == null) {
            return;
        }
        if (!file.exists()) {
            return;
        }
        // 需要保留的文件
        if (!StringUtils.isEmpty(file.getName()) && cookieFileList.contains(file.getName())) {
            return;
        }
        if (file.isFile()) {
            file.delete();
        } else {
            File[] files = file.listFiles();
            if (files == null) {
                return;
            }
            for (File f : files) {
                deleteZhilianCookieFile(f);
            }
            if (!cookieFileList.contains(file.getName())) {
                file.delete();
            }
        }
    }

    public static List<String> getAllFile(File file) {
        List<String> list = new ArrayList<>();
        if (file == null) {
            return list;
        }
        if (!file.exists()) {
            return list;
        }
        if (file.getName().contains("DS_Store")) {
            return list;
        }

        if (file.isFile()) {
            list.add(file.getAbsolutePath());
        } else {
            File[] files = file.listFiles();
            if (files == null) {
                return list;
            }
            for (File f : files) {
                if (f.isFile()) {
                    if (!file.getName().contains("DS_Store")) {
                        list.add(f.getAbsolutePath());
                    }
                } else {
                    list.addAll(getAllFile(f));
                }

            }
        }
        return list;
    }


    /**
     *  获取文件后缀
     * @param key
     * @return
     */
//    public static String getBoosterFilePath (String key) {
//        if (StringUtils.isEmpty(key)) {
//            return "";
//        }
//        String filePath = "";
//        try {
//            // 填充空格转码
//            key = entityDecode(key);
//
//            byte[] cryptData = Base64Util.decode(key);
//            if (null == cryptData) {
//                LOGGER.info("cryptData null");
//            }
//            filePath = YiiUtil.decryptByKey(cryptData, "23b53b4b52ba85ec58fb58ab2253a67f".getBytes(), "");
//
//        } catch (Exception e) {
//            LOGGER.info("获取booster filePath fail :", e);
//        }
//        return filePath;
//    }

//    public static String entityDecode(String key) {
//        if(StringUtil.isNotEmpty(key)) {
//            key = key.replace(" ", "+");
//        }
//        return key;
//    }
//
//    public static String generateKey(String path) throws OperationNotValidException {
//        if (StringUtils.isEmpty(path)) {
//            return null;
//        }
//        String key = "";
//        try {
//            // 生成key
//            byte[] cryptData = YiiUtil.encryptByKey(path.getBytes(), "23b53b4b52ba85ec58fb58ab2253a67f".getBytes(), "");
//            key = Base64Util.encode(cryptData);
//        } catch (Exception e) {
//            LOGGER.info("booster filePath 加密失败:", e);
//        }
//        return key;
//    }

}
